/*
 * Copyright 2006 The Apache Software Foundation Licensed under the Apache
 * License, Version 2.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law
 * or agreed to in writing, software distributed under the License is
 * distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */

package org.apache.ibatis.ibator.api;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;

import org.apache.ibatis.ibator.config.IbatorConfiguration;
import org.apache.ibatis.ibator.config.xml.IbatorConfigurationParser;
import org.apache.ibatis.ibator.exception.InvalidConfigurationException;
import org.apache.ibatis.ibator.exception.XMLParserException;
import org.apache.ibatis.ibator.internal.DefaultShellCallback;
import org.apache.ibatis.ibator.internal.util.messages.Messages;
import org.apache.ibatis.ibator.logging.LogFactory;

/**
 * This class allows Ibator to be run from the command line.
 * 
 * @author Jeff Butler
 */
public class IbatorRunner
{
	private static final String CONFIG_FILE = "-configfile"; //$NON-NLS-1$
	private static final String OVERWRITE = "-overwrite"; //$NON-NLS-1$
	private static final String CONTEXT_IDS = "-contextids"; //$NON-NLS-1$
	private static final String TABLES = "-tables"; //$NON-NLS-1$
	private static final String VERBOSE = "-verbose"; //$NON-NLS-1$
	private static final String FORCE_JAVA_LOGGING = "-forceJavaLogging"; //$NON-NLS-1$
	private static final String HELP_1 = "-?"; //$NON-NLS-1$
	private static final String HELP_2 = "-h"; //$NON-NLS-1$

	public static void main(String[] args)
	{
		if (args.length == 0)
		{
			usage();
			System.exit(0);
			return; // only to satisfy compiler, never returns
		}

		Map<String, String> arguments = parseCommandLine(args);

		if (arguments.containsKey(HELP_1))
		{
			usage();
			System.exit(0);
			return; // only to satisfy compiler, never returns
		}

		if (!arguments.containsKey(CONFIG_FILE))
		{
			writeLine(Messages.getString("RuntimeError.0")); //$NON-NLS-1$
			return;
		}

		List<String> warnings = new ArrayList<String>();

		String configfile = arguments.get(CONFIG_FILE);
		File configurationFile = new File(configfile);
		if (!configurationFile.exists())
		{
			writeLine(Messages.getString("RuntimeError.1", configfile)); //$NON-NLS-1$
			return;
		}

		Set<String> fullyqualifiedTables = new HashSet<String>();
		if (arguments.containsKey(TABLES))
		{
			StringTokenizer st = new StringTokenizer(arguments.get(TABLES), ","); //$NON-NLS-1$
			while (st.hasMoreTokens())
			{
				String s = st.nextToken().trim();
				if (s.length() > 0)
				{
					fullyqualifiedTables.add(s);
				}
			}
		}

		Set<String> contexts = new HashSet<String>();
		if (arguments.containsKey(CONTEXT_IDS))
		{
			StringTokenizer st = new StringTokenizer(
					arguments.get(CONTEXT_IDS), ","); //$NON-NLS-1$
			while (st.hasMoreTokens())
			{
				String s = st.nextToken().trim();
				if (s.length() > 0)
				{
					contexts.add(s);
				}
			}
		}

		try
		{
			IbatorConfigurationParser cp = new IbatorConfigurationParser(
					warnings);
			IbatorConfiguration config = cp
					.parseIbatorConfiguration(configurationFile);

			DefaultShellCallback shellCallback = new DefaultShellCallback(
					arguments.containsKey(OVERWRITE));

			Ibator ibator = new Ibator(config, shellCallback, warnings);

			ProgressCallback progressCallback = arguments.containsKey(VERBOSE) ? new VerboseProgressCallback()
					: null;

			ibator.generate(progressCallback, contexts, fullyqualifiedTables);

		}
		catch (XMLParserException e)
		{
			writeLine(Messages.getString("Progress.3")); //$NON-NLS-1$
			writeLine();
			for (String error : e.getErrors())
			{
				writeLine(error);
			}

			return;
		}
		catch (SQLException e)
		{
			e.printStackTrace();
			return;
		}
		catch (IOException e)
		{
			e.printStackTrace();
			return;
		}
		catch (InvalidConfigurationException e)
		{
			writeLine(Messages.getString("Progress.16")); //$NON-NLS-1$
			for (String error : e.getErrors())
			{
				writeLine(error);
			}
			return;
		}
		catch (InterruptedException e)
		{
			// ignore (will never happen with the DefaultShellCallback)
			;
		}

		for (String warning : warnings)
		{
			writeLine(warning);
		}

		if (warnings.size() == 0)
		{
			writeLine(Messages.getString("Progress.4")); //$NON-NLS-1$
		}
		else
		{
			writeLine();
			writeLine(Messages.getString("Progress.5")); //$NON-NLS-1$
		}
	}

	private static void usage()
	{
		String lines = Messages.getString("Usage.Lines"); //$NON-NLS-1$
		int iLines = Integer.parseInt(lines);
		for (int i = 0; i < iLines; i++)
		{
			String key = "Usage." + i; //$NON-NLS-1$
			writeLine(Messages.getString(key));
		}
	}

	private static void writeLine(String message)
	{
		System.out.println(message);
	}

	private static void writeLine()
	{
		System.out.println();
	}

	private static Map<String, String> parseCommandLine(String[] args)
	{
		List<String> errors = new ArrayList<String>();
		Map<String, String> arguments = new HashMap<String, String>();

		for (int i = 0; i < args.length; i++)
		{
			if (CONFIG_FILE.equalsIgnoreCase(args[i]))
			{
				if ((i + 1) < args.length)
				{
					arguments.put(CONFIG_FILE, args[i + 1]);
				}
				else
				{
					errors.add(Messages.getString(
							"RuntimeError.19", CONFIG_FILE)); //$NON-NLS-1$
				}
				i++;
			}
			else
				if (OVERWRITE.equalsIgnoreCase(args[i]))
				{
					arguments.put(OVERWRITE, "Y"); //$NON-NLS-1$
				}
				else
					if (VERBOSE.equalsIgnoreCase(args[i]))
					{
						arguments.put(VERBOSE, "Y"); //$NON-NLS-1$
					}
					else
						if (HELP_1.equalsIgnoreCase(args[i]))
						{
							arguments.put(HELP_1, "Y"); //$NON-NLS-1$
						}
						else
							if (HELP_2.equalsIgnoreCase(args[i]))
							{
								// put HELP_1 in the map here too - so we only
								// have to check for one entry in the mainline
								arguments.put(HELP_1, "Y"); //$NON-NLS-1$
							}
							else
								if (FORCE_JAVA_LOGGING
										.equalsIgnoreCase(args[i]))
								{
									LogFactory.forceJavaLogging();
								}
								else
									if (CONTEXT_IDS.equalsIgnoreCase(args[i]))
									{
										if ((i + 1) < args.length)
										{
											arguments.put(CONTEXT_IDS,
													args[i + 1]);
										}
										else
										{
											errors.add(Messages
													.getString(
															"RuntimeError.19", CONTEXT_IDS)); //$NON-NLS-1$
										}
										i++;
									}
									else
										if (TABLES.equalsIgnoreCase(args[i]))
										{
											if ((i + 1) < args.length)
											{
												arguments.put(TABLES,
														args[i + 1]);
											}
											else
											{
												errors.add(Messages
														.getString(
																"RuntimeError.19", TABLES)); //$NON-NLS-1$
											}
											i++;
										}
										else
										{
											errors.add(Messages.getString(
													"RuntimeError.20", args[i])); //$NON-NLS-1$
										}
		}

		if (!errors.isEmpty())
		{
			for (String error : errors)
			{
				writeLine(error);
			}

			System.exit(-1);
		}

		return arguments;
	}
}
